package cn.edu.tyu.finance.thingstype.servlet;

import java.io.IOException;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.SQLException;

import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import cn.edu.tyu.util.DbUtil;
import cn.edu.tyu.util.StringUtil;

/**
 * Servlet implementation class AddUserServlet
 */
@WebServlet("/AddUserServlet")
public class AddUserServlet extends HttpServlet {
    private static final long serialVersionUID = 1L;

    /**
     * @see HttpServlet#HttpServlet()
     */
    public AddUserServlet() {
        super();
        // TODO Auto-generated constructor stub
    }

    /**
     * @see HttpServlet#doGet(HttpServletRequest request, HttpServletResponse response)
     */
    protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        //1. 接收网页传来的数据
        String uname = request.getParameter("username");
        String upsw = request.getParameter("userpsw");
        String upsqMD5 = StringUtil.getMD5(upsw);
        //2. 连接数据库把接收到的数据写入表里
        Connection conn = null;
        PreparedStatement pstmt = null;
        conn = DbUtil.getConnection();
        int flag = 0;
        String sql = "INSERT INTO users VALUES(?,?,?)";
        try {
            pstmt = conn.prepareStatement(sql);
            pstmt.setString(1, (int)(Math.random()*1000) + "");
            pstmt.setString(2, uname);
            pstmt.setString(3, upsw);
            flag = pstmt.executeUpdate();//下达执行指令
            if (flag == 0) {
                response.sendRedirect("fail.jsp");
            } else {
                response.sendRedirect("ok.jsp");
            }
        } catch (SQLException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            //3. 关闭数据库资源
            DbUtil.backConn(conn, pstmt, null);
        }
    }

    /**
     * @see HttpServlet#doPost(HttpServletRequest request, HttpServletResponse response)
     */
    protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        // TODO Auto-generated method stub
        doGet(request, response);
    }

}

